tensor([[0.5600, 0.5600, 0.5600, 0.5600, 0.5600],
        [0.4400, 0.4400, 0.4400, 0.4400, 0.4400]], grad_fn=<RoundBackward1>)